/*
 * Copyright (C) 2025 Huawei Device Co., Ltd.
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
#include <napi/native_api.h>
#include <string>
#include <ohos_utils/ohos_log.h>
#include "ohos_utils/napi_handler.h"
#include <openssl/des.h>
#include <openssl/evp.h>

static const size_t BLOCK_SIZE = 8;

static napi_value desDecrypt(napi_env env, napi_callback_info info)
{
    NapiHandler napiHandler(env, info, PARAM_COUNT_2);
    ArrayBuffer data = napiHandler.ParseArgAs<ArrayBuffer>(INDEX_0);
    ArrayBuffer key = napiHandler.ParseArgAs<ArrayBuffer>(INDEX_1);
    if (data.size() != BLOCK_SIZE || key.size() != BLOCK_SIZE) {
        LOGE("length is error");
        return nullptr;
    }

    DES_key_schedule schedule;
    DES_cblock desKey;
    DES_cblock input;
    DES_cblock output;

    std::memcpy(desKey, key.data(), BLOCK_SIZE);
    DES_set_key(&desKey, &schedule);
    std::memcpy(input, data.data(), BLOCK_SIZE);
    DES_ecb_encrypt(&input, &output, &schedule, DES_DECRYPT);

    return napiHandler.GetNapiValue<ArrayBuffer>(std::vector<uint8_t>(output, output + BLOCK_SIZE));
}

static napi_value desEncrypt(napi_env env, napi_callback_info info)
{
    NapiHandler napiHandler(env, info, PARAM_COUNT_2);
    ArrayBuffer data = napiHandler.ParseArgAs<ArrayBuffer>(INDEX_0);
    ArrayBuffer key = napiHandler.ParseArgAs<ArrayBuffer>(INDEX_1);
    if (data.size() != BLOCK_SIZE || key.size() != BLOCK_SIZE) {
        LOGE("length is error");
        return nullptr;
    }

    DES_key_schedule schedule;
    DES_cblock desKey;
    DES_cblock input;
    DES_cblock output;

    std::memcpy(desKey, key.data(), BLOCK_SIZE);
    DES_set_key(&desKey, &schedule);
    std::memcpy(input, data.data(), BLOCK_SIZE);
    DES_ecb_encrypt(&input, &output, &schedule, DES_ENCRYPT);

    return napiHandler.GetNapiValue<ArrayBuffer>(std::vector<uint8_t>(output, output + BLOCK_SIZE));
}

EXTERN_C_START
static napi_value Init(napi_env env, napi_value exports)
{
    // YYEVAComponent::instance().initComponent(env, exports);
    napi_property_descriptor desc[] = {
        DECLARE_NAPI_FUNCTION("desDecrypt", desDecrypt), DECLARE_NAPI_FUNCTION("desEncrypt", desEncrypt)};
    NAPI_CALL(env, napi_define_properties(env, exports, sizeof(desc) / sizeof(desc[0]), desc));
    return exports;
}
EXTERN_C_END

static napi_module nativeModule = {
    .nm_version = 1,
    .nm_flags = 0,
    .nm_filename = nullptr,
    .nm_register_func = Init,
    .nm_modname = "openssl_napi",
    .nm_priv = ((void *)0),
    .reserved = {0},
};

extern "C" __attribute__((constructor)) void RegisterModule(void)
{
    napi_module_register(&nativeModule);
}